/* Copyright (c) 2012 Nordic Semiconductor. All Rights Reserved.
 *
 * The information contained herein is confidential property of Nordic
 * Semiconductor ASA.Terms and conditions of usage are described in detail
 * in NORDIC SEMICONDUCTOR STANDARD SOFTWARE LICENSE AGREEMENT.
 *
 * Licensees are granted free, non-transferable use of the information. NO
 * WARRANTY of ANY KIND is provided. This heading must NOT be removed from
 * the file.
 *
 * $$
 */

/** @file
 *
 * @defgroup ble_sdk_app_template_main main.c
 * @{
 * @ingroup ble_sdk_app_template
 * @brief Template project main file.
 *
 * This file contains a template for creating a new application. It has the code necessary to wakeup from button, advertise, get a connection
 * restart advertising on disconnect and if no new connection created go back to system-off mode.
 * It can easily be used as a starting point for creating a new application, the comments identified with 'YOUR_JOB' indicates where
 * and how you can customize.
 */

#include <stdint.h>
#include <string.h>
#include "nordic_common.h"
#include "nrf.h"
#include "nrf_assert.h"
#include "nrf_error.h"
#include "nrf_gpio.h"
#include "nrf51_bitfields.h"
#include "ble.h"
#include "ble_hci.h"
#include "ble_srv_common.h"
#include "ble_advdata.h"
#include "ble_conn_params.h"
#include "ble_nrf6310_pins.h"
#include "app_scheduler.h"
#include "ble_stack_handler.h"
#include "app_timer.h"


#define DEVICE_NAME                     "NORDIC_TEMPLATE"                           /**< Name of device. Will be included in the advertising data. */

#define APP_ADV_INTERVAL                64                                          /**< The advertising interval (in units of 0.625 ms. This value corresponds to 40 ms). */
#define APP_ADV_TIMEOUT_IN_SECONDS      180                                         /**< The advertising timeout (in units of seconds). */

#define MIN_CONN_INTERVAL               ((1 * 800) / 2)                             /**< Minimum connection interval (0.5 seconds). */
#define MAX_CONN_INTERVAL               (1 * 800)                                   /**< Maximum connection interval (1 second). */
#define SLAVE_LATENCY                   0                                           /**< Slave latency. */
#define CONN_SUP_TIMEOUT                (4 * 100)                                   /**< Connection supervisory timeout (4 seconds). */
#define FIRST_CONN_PARAMS_UPDATE_DELAY  (20 * 1000)                                 /**< Time from initiating event (connect or start of notification) to first time ble_gap_conn_param_update is called (15 seconds). */
#define NEXT_CONN_PARAMS_UPDATE_DELAY   (5 * 1000)                                  /**< Time between each call to ble_gap_conn_param_update after the first (5 seconds). */
#define MAX_CONN_PARAMS_UPDATE_COUNT    3                                           /**< Number of attempts before giving up the connection parameter negotiation. */

#define SEC_PARAM_TIMEOUT               60                                          /**< Timeout for Pairing Request or Security Request (in seconds). */
#define SEC_PARAM_BOND                  1                                           /**< Perform bonding. */
#define SEC_PARAM_MITM                  0                                           /**< Man In The Middle protection not required. */
#define SEC_PARAM_IO_CAPABILITIES       BLE_GAP_IO_CAPS_NONE                        /**< No I/O capabilities. */
#define SEC_PARAM_OOB                   0                                           /**< Out Of Band data not available. */
#define SEC_PARAM_MIN_KEY_SIZE          7                                           /**< Minimum encryption key size. */
#define SEC_PARAM_MAX_KEY_SIZE          16                                          /**< Maximum encryption key size. */

static ble_gap_sec_params_t             s_sec_params;                               /**< Security requirements for this application. */
static uint16_t                         s_conn_handle = BLE_CONN_HANDLE_INVALID;    /**< Handle of the current connection. */


/**@brief Assert macro callback function.
 *
 * @details This function will be called if the ASSERT macro fails.
 *
 * @param[in]   line_num   Line number of the failing ASSERT call.
 * @param[in]   file_name  File name of the failing ASSERT call.
 */
void assert_nrf_callback(uint16_t line_num, const uint8_t * file_name)
{
    // Copying parameters to static variables because parameters are not accessible in debugger
    static volatile uint8_t  s_file_name[128];
    static volatile uint16_t s_line_num;

    strcpy((char *)s_file_name, (const char *)file_name);
    s_line_num = line_num;
    UNUSED_VARIABLE(s_file_name);
    UNUSED_VARIABLE(s_line_num);
    
    nrf_gpio_pin_set(ASSERT_LED_PIN_NO);

    for (;;)
    {
        // Loop forever. On assert, the system can only recover on reset
    }
}


/**@brief Service error handler.
 *
 * @details A pointer to this function will be passed to each service which may need to inform the 
 *          application about an error.
 *
 * @param[in]   nrf_error   Error code containing information about what went wrong.
 */
/*
// YOUR_JOB: Uncomment this function and make it handle error situations sent back to your 
//           application by the services it uses.
static void service_error_handler(uint32_t nrf_error)
{
    ASSERT(false);
} */


/**@brief LEDs initialization.
 *
 * @details Initializes all LEDs used by the application.
 */
static void leds_init(void)
{
    GPIO_LED_CONFIG(ADVERTISING_LED_PIN_NO);
    GPIO_LED_CONFIG(CONNECTED_LED_PIN_NO);
    GPIO_LED_CONFIG(ASSERT_LED_PIN_NO);
    
    // YOUR_JOB: Add additional LED initialiazations if needed.
}


/**@brief Timer initialization.
 *
 * @details Initializes the timer module.
 */
static void timers_init(void)
{
    uint32_t err_code;
    
    // Initialize timer module, making it use the scheduler
    err_code = app_timer_init(app_sched_timer_event_schedule);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief GAP initialization.
 *
 * @details This function shall be used to setup all the necessary GAP (Generic Access Profile) 
 *          parameters of the device. It also sets the permissions and appearance.
 */
static void gap_params_init(void)
{
    uint32_t                err_code;
    ble_gap_conn_params_t   gap_conn_params;
    ble_gap_conn_sec_mode_t sec_mode;

    BLE_GAP_CONN_SEC_MODE_SET_OPEN(&sec_mode);
    
    err_code = ble_gap_device_name_set(&sec_mode, DEVICE_NAME);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }

    /* YOUR_JOB: Use an appearance value matching the application's use case.
    err_code = ble_gap_appearance_set(BLE_APPEARANCE_);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }*/
    
    memset(&gap_conn_params, 0, sizeof(gap_conn_params));

    gap_conn_params.min_conn_interval = MIN_CONN_INTERVAL;
    gap_conn_params.max_conn_interval = MAX_CONN_INTERVAL;
    gap_conn_params.slave_latency     = SLAVE_LATENCY;
    gap_conn_params.conn_sup_timeout  = CONN_SUP_TIMEOUT;

    err_code = ble_gap_ppcp_set(&gap_conn_params);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Advertising functionality initialization.
 *
 * @details Encodes the required advertising data and passes it to the stack.
 *          Also builds a structure to be passed to the stack when starting advertising.
 */
static void advertising_init(void)
{
    uint32_t      err_code;
    ble_advdata_t advdata;
    uint8_t       flags = BLE_GAP_DISC_MODE_LIMITED | BLE_GAP_DISC_BR_EDR_NOT_SUPPORTED;
    
    // YOUR_JOB: Use UUIDs for service(s) used in your application.
    ble_uuid_t adv_uuids[] = {{BLE_UUID_BATTERY_SERVICE, BLE_UUID_TYPE_BLE}};

    // Build and set advertising data
    memset(&advdata, 0, sizeof(advdata));
    
    advdata.name_type               = BLE_ADVDATA_FULL_NAME;
    advdata.include_appearance      = true;
    advdata.flags.size              = sizeof(flags);
    advdata.flags.p_data            = &flags;
    advdata.uuids_complete.uuid_cnt = sizeof(adv_uuids) / sizeof(adv_uuids[0]);
    advdata.uuids_complete.p_uuids  = adv_uuids;
    
    err_code = ble_advdata_set(&advdata, NULL);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Initialize services that will be used by the application.
 */
static void services_init(void)
{
    // YOUR_JOB: Add code to initialize the services used by the application.
}


/**@brief Initialize security parameters.
 */
static void sec_params_init(void)
{
    s_sec_params.timeout      = SEC_PARAM_TIMEOUT;
    s_sec_params.bond         = SEC_PARAM_BOND;
    s_sec_params.mitm         = SEC_PARAM_MITM;
    s_sec_params.io_caps      = SEC_PARAM_IO_CAPABILITIES;
    s_sec_params.oob          = SEC_PARAM_OOB;  
    s_sec_params.min_key_size = SEC_PARAM_MIN_KEY_SIZE;
    s_sec_params.max_key_size = SEC_PARAM_MAX_KEY_SIZE;
}


/**@brief Connection Parameters Module handler.
 *
 * @details This function will be called for all events in the Connection Parameters Module which
 *          are passed to the application.
 *          @note All this function does is to disconnect. This could have been done by simply
 *                setting the disconnect_on_fail config parameter, but instead we use the event
 *                handler mechanism to demonstrate its use.
 *
 * @param[in]   p_evt   Event received from the Connection Parameters Module.
 */
static void on_conn_params_evt(ble_conn_params_evt_t * p_evt)
{
    uint32_t err_code;
    
    ASSERT(p_evt->evt_type == BLE_CONN_PARAMS_EVT_FAILED);
    
    err_code = ble_gap_disconnect(s_conn_handle, BLE_HCI_CONN_INTERVAL_UNACCEPTABLE);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Connection Parameters module error handler.
 *
 * @param[in]   nrf_error   Error code containing information about what went wrong.
 */
static void conn_params_error_handler(uint32_t nrf_error)
{
    ASSERT(false);
}


/**@brief Initialize the Connection Parameters module.
 */
static void conn_params_init(void)
{
    uint32_t               err_code;
    ble_conn_params_init_t cp_init;
    
    memset(&cp_init, 0, sizeof(cp_init));

    cp_init.p_conn_params                  = NULL;
    cp_init.first_conn_params_update_delay = FIRST_CONN_PARAMS_UPDATE_DELAY;
    cp_init.next_conn_params_update_delay  = NEXT_CONN_PARAMS_UPDATE_DELAY;
    cp_init.max_conn_params_update_count   = MAX_CONN_PARAMS_UPDATE_COUNT;
    cp_init.start_on_notify_cccd_handle    = BLE_GATTS_HANDLE_INVALID;
    cp_init.disconnect_on_fail             = false;
    cp_init.evt_handler                    = on_conn_params_evt;
    cp_init.error_handler                  = conn_params_error_handler;
    
    err_code = ble_conn_params_init(&cp_init);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Start timers.
*/
static void timers_start(void)
{
    /* YOUR_JOB: Start your timers. below is an example of how to start a timer.
    uint32_t err_code;
    
    err_code = app_timer_start(APP_TIMER_ID, 
                               timer_timeout_handler, 
                               TIMER_INTERVAL, 
                               APP_TIMER_MODE_REPEATED/APP_TIMER_MODE_SINGLE_SHOT, 
                               APP_SCHED_QUEUE_LOW_PRI);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    } */
}


/**@brief Start advertising.
 */
static void advertising_start(void)
{
    uint32_t             err_code;
    ble_gap_adv_params_t adv_params;
    
    // Start advertising
    memset(&adv_params, 0, sizeof(adv_params));
    
    adv_params.type        = BLE_GAP_ADV_TYPE_ADV_IND;
    adv_params.p_peer_addr = NULL;
    adv_params.fp          = BLE_GAP_ADV_FP_ANY;
    adv_params.interval    = APP_ADV_INTERVAL;
    adv_params.timeout     = APP_ADV_TIMEOUT_IN_SECONDS;

    err_code = ble_gap_adv_start(&adv_params);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
    nrf_gpio_pin_set(ADVERTISING_LED_PIN_NO);
}


/**@brief Application's BLE Stack event handler.
 *
 * @param[in]   p_ble_evt   Bluetooth stack event.
 */
static void on_ble_evt(ble_evt_t * p_ble_evt)
{
    uint32_t                         err_code = NRF_SUCCESS;
    static ble_gap_evt_auth_status_t s_auth_status;
    ble_gap_enc_info_t *             p_enc_info;
    
    switch (p_ble_evt->header.evt_id)
    {
        case BLE_GAP_EVT_CONNECTED:
            nrf_gpio_pin_set(CONNECTED_LED_PIN_NO);
            nrf_gpio_pin_clear(ADVERTISING_LED_PIN_NO);
            app_timer_stop(APP_TIMER_STOP_ADVERTISE);
            s_conn_handle = p_ble_evt->evt.gatts_evt.conn_handle;
            break;
            
        case BLE_GAP_EVT_DISCONNECTED:
            nrf_gpio_pin_clear(CONNECTED_LED_PIN_NO);
            s_conn_handle = BLE_CONN_HANDLE_INVALID;
            advertising_start();
            break;
            
        case BLE_GAP_EVT_SEC_PARAMS_REQUEST:
            err_code = ble_gap_sec_params_reply(s_conn_handle, 
                                                BLE_GAP_SEC_STATUS_SUCCESS, 
                                                &s_sec_params);
            break;
            
        case BLE_GATTS_EVT_SYS_ATTR_MISSING:
            err_code = ble_gatts_sys_attr_set(s_conn_handle, NULL, 0);
            break;

        case BLE_GAP_EVT_AUTH_STATUS:
            s_auth_status = p_ble_evt->evt.gap_evt.params.auth_status;
            break;
            
        case BLE_GAP_EVT_SEC_INFO_REQUEST:
            p_enc_info = &s_auth_status.periph_keys.enc_info;
            if (p_enc_info->div == p_ble_evt->evt.gap_evt.params.sec_info_request.div)
            {
                err_code = ble_gap_sec_info_reply(s_conn_handle, p_enc_info, NULL);
            }
            else
            {
                // No keys found for this device
                err_code = ble_gap_sec_info_reply(s_conn_handle, NULL, NULL);
            }
            break;

        case BLE_GAP_EVT_TIMEOUT:
            if (p_ble_evt->evt.gap_evt.params.timeout.src == BLE_GAP_TIMEOUT_TYPE_ADVERTISEMENT)
            { 
                nrf_gpio_pin_clear(ADVERTISING_LED_PIN_NO);

                // Go to system-off mode (this function will not return; wakeup will cause a reset)
                err_code = nrf_power_system_off();    
            }
            break;

        default:
            break;
    }

    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Dispatches a BLE stack event to all modules with a BLE stack event handler.
 *
 * @details This function is called from the scheduler in the main loop after a BLE stack
 *          event has been received.
 *
 * @param[in]   p_ev_data   Scheduler event containing a bluetooth event.
 */
static void ble_evt_dispatch(app_sched_event_data_t * p_ev_data)
{
    ble_evt_t * p_ble_evt;
    
    ASSERT(p_ev_data->event_id == APP_SCHED_EVENT_BLE);
    
    p_ble_evt = &p_ev_data->data.ble.ble_evt;

    on_ble_evt(p_ble_evt);
    ble_conn_params_on_ble_evt(p_ble_evt);
    /* 
    YOUR_JOB: Add service ble_evt handlers calls here, like, for example:
    ble_bas_on_ble_evt(&s_bas, p_ble_evt);
    */
}


/**@brief Pass bluetooth stack event to scheduler.
 *
 * @details This function is called from the BLE Stack event interrupt handler after a BLE stack
 *          event has been received.
 *
 * @param[in]   p_ble_evt   Bluetooth stack event.
 */
static void ble_evt_schedule(ble_evt_t * p_ble_evt)
{
    uint32_t               err_code;
    app_sched_event_data_t ev_data;
    
    ev_data.event_id         = APP_SCHED_EVENT_BLE;
    ev_data.data.ble.ble_evt = *p_ble_evt;
    
    err_code = app_sched_event_put(&ev_data, ble_evt_dispatch, APP_SCHED_QUEUE_LOW_PRI);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief BLE stack initialization.
 *
 * @details Initializes the SoftDevice and the BLE event interrupt.
 */
static void ble_stack_init(void)
{
    uint32_t err_code = ble_stack_handler_init(NRF_CLOCK_LFCLKSRC_XTAL_20_PPM, ble_evt_schedule);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief GPIOTE interrupt handler.
 */
void GPIOTE_IRQHandler(void)
{
  // YOUR_JOB: Add code to detect which IO has generated the interrupt and forward the corresponding
  //           event to the scheduler or handle it in any other way. Refer to the example 
  //           applications to see how the code here may look like.
}


/**@brief Initialize GPIOTE module for detecting buttons.
 */
static void buttons_init(void)
{
    uint32_t err_code;
    uint32_t per_rdy;
    
    // Power on GPIOTE module
    err_code = nrf_power_perpower_set(POWER_PERPOWER_GPIOTE_Msk);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }

    do
    {
        err_code = nrf_power_perrdy_get(&per_rdy);
        if (err_code != NRF_SUCCESS)
        {
            ASSERT(false);
        }
    } while ((per_rdy & POWER_PERRDY_GPIOTE_Msk) == 0);
    
    // Initialize GPIOTE module
    NRF_GPIOTE->INTENCLR = 0xffffffffUL;
    /* YOUR_JOB: Uncomment this part if you need to receive PORT event (that occurs during
                 button presses for example). Appropriate handling of the PORT event should be done 
                 in GPIOTE_IRQHandler. Refer to the source code of example applications for usage.
    NRF_GPIOTE->INTENSET = GPIOTE_INTENSET_PORT_Msk;
    */
    
    // Set Button 0 as wakeup source
    GPIO_PIN_CONFIG(NRF6310_BUTTON_0, 
                    GPIO_PIN_CNF_DIR_Input, 
                    GPIO_PIN_CNF_PULL_Disabled, 
                    GPIO_PIN_CNF_DRIVE_S0S1, 
                    GPIO_PIN_CNF_SENSE_Low);
    
    // Enable GPIOTE interrupt
    err_code = nrf_nvic_ClearPendingIRQ(GPIOTE_IRQn);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
    
    err_code = nrf_nvic_SetPriority(GPIOTE_IRQn, NRF_APP_PRIORITY_LOW);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
    
    err_code = nrf_nvic_EnableIRQ(GPIOTE_IRQn);
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Power manager.
 */
static void power_manage(void)
{
    uint32_t err_code = nrf_wait_for_app_event();
    if (err_code != NRF_SUCCESS)
    {
        ASSERT(false);
    }
}


/**@brief Application main function.
 */
int main(void)
{
    // Initialize
    leds_init();
    ble_stack_init();
    app_sched_init();
    timers_init();
    buttons_init();
    gap_params_init();
    advertising_init();
    services_init();
    conn_params_init();
    sec_params_init();
    
    // Start execution
    timers_start();
    advertising_start();
    
    // Enter main loop
    for (;;)
    {
        app_sched_schedule();
        power_manage();
    }
}

/** 
 * @}
 */
